CUDA=cu101
TORCH=1.7.1
prox pip3 install --no-cache-dir --no-index torch-scatter -f https://data.pyg.org/whl/torch-${TORCH}+${CUDA}.html && \
prox pip3 install --no-cache-dir --no-index torch-sparse -f https://data.pyg.org/whl/torch-${TORCH}+${CUDA}.html && \
prox pip3 install torch-geometric